# -*- coding: utf-8 -*-
# !/usr/bin/env python
"""
-------------------------------------------------
   File Name：     2onnx
   Description :   
   Author :       lth
   date：          2022/12/7
-------------------------------------------------
   Change Activity:
                   2022/12/7 12:07: create this script
-------------------------------------------------
"""
__author__ = 'lth'

import torch.onnx

from model import NeRF


dummy_input = torch.randn(1,90).cuda()
model = NeRF().cuda()
torch.onnx.export(model,dummy_input,"NeRF.onnx",verbose=True)